package com.dh.leetcode.four;

import org.junit.Assert;
import org.junit.Test;

/**
 * @ClassName: _441_arranging_coins
 * @Description: 441. 排列硬币
 * 你总共有 n 枚硬币，并计划将它们按阶梯状排列。对于一个由 k 行组成的阶梯，其第 i 行必须正好有 i 枚硬币。阶梯的最后一行 可能 是不完整的。
 * <p>
 * 给你一个数字 n ，计算并返回可形成 完整阶梯行 的总行数。
 * <p>
 * 输入：n = 5
 * 输出：2
 * 解释：因为第三行不完整，所以返回 2 。
 * <p>
 * 输入：n = 8
 * 输出：3
 * 解释：因为第四行不完整，所以返回 3 。
 * <p>
 * 提示：
 * 1 <= n <= 231 - 1
 *
 * https://leetcode-cn.com/problems/arranging-coins/
 *
 * @Author: shouzimu
 * @Date: 2021/10/11 9:43
 */
public class _441_arranging_coins {

    /**
     * 1+2+3+...+n = n(n+1)/2
     * <p>
     * 所以问题转换为求平方根
     * <p>
     *
     * @param n
     * @return
     */
    public int arrangeCoins(int n) {
        long mid = 0, res = 0, left = 0, right = n;
        while (left <= right) {
            mid = left + (right - left) / 2;
            if ((mid * (mid + 1) / 2 <= n)) {
                left = mid + 1;
                res = mid;
            } else {
                right = mid - 1;
            }
        }
        return (int) res;
    }


    @Test
    public void arrangeCoinsTest() {
        Assert.assertEquals(60070, arrangeCoins(1804289383));
        Assert.assertEquals(1, arrangeCoins(1));
        Assert.assertEquals(1, arrangeCoins(2));
        Assert.assertEquals(2, arrangeCoins(5));
        Assert.assertEquals(3, arrangeCoins(8));
        Assert.assertEquals(4, arrangeCoins(10));
        Assert.assertEquals(5, arrangeCoins(16));
    }
}
